package com.example.leetcode.leetcode.hot100;

/**
 * @author haitao
 * @date 2021/11/9 2:47 下午
 * 给定一个整数数组，其中第i个元素代表了第i天的股票价格 。
 * 设计一个算法计算出最大利润。在满足以下约束条件下，你可以尽可能地完成更多的交易（多次买卖一支股票）:
 * 你不能同时参与多笔交易（你必须在再次购买前出售掉之前的股票）。
 * 卖出股票后，你无法在第二天买入股票 (即冷冻期为 1 天)。
 * 示例:
 * 输入: [1,2,3,0,2]
 * 输出: 3
 * 解释: 对应的交易状态为: [买入, 卖出, 冷冻期, 买入, 卖出]
 * 动态规划（dp[][3]）求最佳买卖股票时机含冻结期
 * <p>
 * 那么对于求第i天最大收益有三种情况
 * 1.当天持有一张股票dp[i][0]
 * dp[i][0] = Max(dp[i-1][0],dp[i-1][2]-price[i])
 * 2.当天不持有股票，但是在冻结期当中dp[i][1]
 * dp[i][1] = dp[i-1][0] + price[i]  昨天持有股票卖掉
 * 3.当天不持有股票，不再冻结期当中dp[i][2]
 * dp[i][0] = Max(dp[i-1][1],dp[i-1][2])
 * 最终结果
 * d[i]=Max(dp[i][0],dp[i][1],dp[i][2])，由于dp[i][0]持有股票，无意义，因此只需要求Max(dp[i][1],dp[i][2])
 */
public class Solution309 {
    public int maxProfit(int[] prices) {
        int n = prices.length;
        if (n == 0) {
            return 0;
        }
        int[][] dp = new int[n][3];
        dp[0][0] = -prices[0];
        for (int i = 1; i < n; i++) {
            dp[i][0] = Math.max(dp[i - 1][0], dp[i - 1][2] - prices[i]);
            dp[i][1] = dp[i - 1][0] + prices[i];
            dp[i][2] = Math.max(dp[i - 1][1], dp[i - 1][2]);
        }
        return Math.max(dp[n - 1][1], dp[n - 1][2]);
    }

    public static void main(String[] args) {
        int[] prices = new int[]{1, 2, 3, 0, 2};
        Solution309 solution309 = new Solution309();
        System.out.println(solution309.maxProfit(prices));
    }
}
